import os
import logging
from argparse import ArgumentParser

from agentverse.logging import logger
from agentverse.simulation import Simulation


def run_one_time(query_type: int, info: dict):
    tasks_dir = os.path.join(os.path.dirname(__file__), "..", "AgentVerse", "agentverse", "tasks")
    agentverse = Simulation.from_task("simulation/meituan", tasks_dir)
    response = agentverse.run_one_time(query_type, info)
    return response


def get_response_from_llm(query_type: int, info: dict):
    ans = run_one_time(query_type, info)
    response = ans[0].content
    return response
